import numpy as np  
import matplotlib.pyplot as plt   
  
plt.figure(1) # 创建图表1  
plt.figure(2) # 创建图表2  
ax1 = plt.subplot(211) # 在图表2中创建子图1  
ax2 = plt.subplot(212) # 在图表2中创建子图2  

def fx(x,i):
    return i*x*x
   
x = np.linspace(0, 3, 100)  
for i in range(5):  
    plt.figure(1)  #选择图表1  
    plt.plot(x, fx(x,i))  
    plt.sca(ax1)   #选择图表2的子图1  
    plt.plot(x, np.sin(i*x))  
    plt.sca(ax2)  # 选择图表2的子图2  
    plt.plot(x, np.cos(i*x))  
plt.show() 
